from django.http import HttpResponseRedirect
from legendofkrouzt.item.models import Item, PersonagemItem, Arma, \
    PersonagemArma, Armadura, PersonagemArmadura
from legendofkrouzt.personagem.models import Personagem
from legendofkrouzt.views import TemplatePersonagem

def index(request):
    loja = Item.objects.all()
    if(request.method == 'POST'):
        item = Item.objects.get(pk=request.POST['idItem'])
        personagem = Personagem.objects.get(pk=request.session['id_Personagem'])
        if(personagem.ouro >= item.preco):
            itemPersonagem = PersonagemItem()
            itemPersonagem.item = item
            itemPersonagem.personagem = personagem
            itemPersonagem.nome = item.nome
            itemPersonagem.descricao = item.descricao
            itemPersonagem.agilidade = item.agilidade
            itemPersonagem.constituicao = item.constituicao
            itemPersonagem.inteligencia = item.inteligencia
            itemPersonagem.forca = item.forca
            itemPersonagem.hp = item.hp
            itemPersonagem.mp = item.mp
            itemPersonagem.nivel = item.nivel
            itemPersonagem.img = item.img
            itemPersonagem.preco_venda = (item.preco-(item.preco*0.3))
            personagem.ouro -= item.preco
            personagem.save()
            itemPersonagem.save()
            sucess = "Compra realizada com sucesso!"
        else:
            notsucess = "Voce nao tem ouro suficiente para comprar esse item!"
    return TemplatePersonagem(request, 'loja/index.html', locals())

def armas(request):
    armas = Arma.objects.all()
    if(request.method == 'POST'):
        arma = Arma.objects.get(pk=request.POST['idArma'])
        personagem = Personagem.objects.get(pk=request.session['id_Personagem'])
        if(personagem.ouro >= arma.preco):
            armaPersonagem = PersonagemArma()
            armaPersonagem.arma = arma
            armaPersonagem.personagem = personagem
            armaPersonagem.nome = arma.nome
            armaPersonagem.descricao = arma.descricao
            armaPersonagem.agilidade = arma.agilidade
            armaPersonagem.constituicao = arma.constituicao
            armaPersonagem.inteligencia = arma.inteligencia
            armaPersonagem.forca = arma.forca
            armaPersonagem.hp = arma.hp
            armaPersonagem.mp = arma.mp
            armaPersonagem.nivel = arma.nivel
            armaPersonagem.img = arma.img
            armaPersonagem.tipo_equipamento = arma.tipo_equipamento
            armaPersonagem.preco_venda = (arma.preco-(arma.preco*0.3))
            personagem.ouro -= arma.preco
            personagem.save()
            armaPersonagem.save()
            sucess = "Compra realizada com sucesso!"
        else:
            notsucess = "Voce nao tem ouro suficiente para comprar esse item!"
    return TemplatePersonagem(request, 'loja/armas.html', locals())

def armaduras(request):
    armaduras = Armadura.objects.all()
    if(request.method == 'POST'):
        armadura = Armadura.objects.get(pk=request.POST['idArmadura'])
        personagem = Personagem.objects.get(pk=request.session['id_Personagem'])
        if(personagem.ouro >= armadura.preco):
            armaduraPersonagem = PersonagemArmadura()
            armaduraPersonagem.armadura = armadura
            armaduraPersonagem.personagem = personagem
            armaduraPersonagem.nome = armadura.nome
            armaduraPersonagem.descricao = armadura.descricao
            armaduraPersonagem.agilidade = armadura.agilidade
            armaduraPersonagem.constituicao = armadura.constituicao
            armaduraPersonagem.inteligencia = armadura.inteligencia
            armaduraPersonagem.forca = armadura.forca
            armaduraPersonagem.hp = armadura.hp
            armaduraPersonagem.mp = armadura.mp
            armaduraPersonagem.nivel = armadura.nivel
            armaduraPersonagem.img = armadura.img
            armaduraPersonagem.tipo_equipamento = armadura.tipo_equipamento
            armaduraPersonagem.preco_venda = (armadura.preco-(armadura.preco*0.3))
            personagem.ouro -= armadura.preco
            personagem.save()
            armaduraPersonagem.save()
            sucess = "Compra realizada com sucesso!"
        else:
            notsucess = "Voce nao tem ouro suficiente para comprar esse item!"
    return TemplatePersonagem(request, 'loja/armaduras.html', locals())